import pickle
import numpy as np
import pathlib

label = np.load('/root/lmh/code/CTF/utils_code/label_yidong.npy')
label_pkl_dir = pathlib.Path('/root/lmh/code/CTF/label_result')
for m in range(label.shape[0]):
    with open(label_pkl_dir / f'{m}.pkl', mode='wb') as f:
        pickle.dump(label[m].tolist(), f)